home *** CD-ROM | disk | FTP | other *** search
/ Chip 2007 January, February, March & April / Chip-Cover-CD-2007-02.iso / Pakiet bezpieczenstwa / mini Pentoo LiveCD 2006.1 / mpentoo-2006.1.iso / livecd.squashfs / usr / lib / python2.4 / test / test_os.pyo (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2005-10-18  |  10.3 KB  |  362 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyo (Python 2.4)
  3.  
  4. import os
  5. import unittest
  6. import warnings
  7. from test import test_support
  8. warnings.filterwarnings('ignore', 'tempnam', RuntimeWarning, __name__)
  9. warnings.filterwarnings('ignore', 'tmpnam', RuntimeWarning, __name__)
  10.  
  11. class TemporaryFileTests(unittest.TestCase):
  12.     
  13.     def setUp(self):
  14.         self.files = []
  15.         os.mkdir(test_support.TESTFN)
  16.  
  17.     
  18.     def tearDown(self):
  19.         for name in self.files:
  20.             os.unlink(name)
  21.         
  22.         os.rmdir(test_support.TESTFN)
  23.  
  24.     
  25.     def check_tempfile(self, name):
  26.         self.failIf(os.path.exists(name), 'file already exists for temporary file')
  27.         open(name, 'w')
  28.         self.files.append(name)
  29.  
  30.     
  31.     def test_tempnam(self):
  32.         if not hasattr(os, 'tempnam'):
  33.             return None
  34.         
  35.         warnings.filterwarnings('ignore', 'tempnam', RuntimeWarning, 'test_os$')
  36.         self.check_tempfile(os.tempnam())
  37.         name = os.tempnam(test_support.TESTFN)
  38.         self.check_tempfile(name)
  39.         name = os.tempnam(test_support.TESTFN, 'pfx')
  40.         self.assert_(os.path.basename(name)[:3] == 'pfx')
  41.         self.check_tempfile(name)
  42.  
  43.     
  44.     def test_tmpfile(self):
  45.         if not hasattr(os, 'tmpfile'):
  46.             return None
  47.         
  48.         fp = os.tmpfile()
  49.         fp.write('foobar')
  50.         fp.seek(0, 0)
  51.         s = fp.read()
  52.         fp.close()
  53.         self.assert_(s == 'foobar')
  54.  
  55.     
  56.     def test_tmpnam(self):
  57.         import sys
  58.         if not hasattr(os, 'tmpnam'):
  59.             return None
  60.         
  61.         warnings.filterwarnings('ignore', 'tmpnam', RuntimeWarning, 'test_os$')
  62.         name = os.tmpnam()
  63.         if sys.platform in ('win32',):
  64.             self.failIf(os.path.exists(name), 'file already exists for temporary file')
  65.         else:
  66.             self.check_tempfile(name)
  67.  
  68.  
  69.  
  70. class StatAttributeTests(unittest.TestCase):
  71.     
  72.     def setUp(self):
  73.         os.mkdir(test_support.TESTFN)
  74.         self.fname = os.path.join(test_support.TESTFN, 'f1')
  75.         f = open(self.fname, 'wb')
  76.         f.write('ABC')
  77.         f.close()
  78.  
  79.     
  80.     def tearDown(self):
  81.         os.unlink(self.fname)
  82.         os.rmdir(test_support.TESTFN)
  83.  
  84.     
  85.     def test_stat_attributes(self):
  86.         if not hasattr(os, 'stat'):
  87.             return None
  88.         
  89.         import stat
  90.         result = os.stat(self.fname)
  91.         self.assertEquals(result[stat.ST_SIZE], 3)
  92.         self.assertEquals(result.st_size, 3)
  93.         import sys
  94.         members = dir(result)
  95.         for name in dir(stat):
  96.             if name[:3] == 'ST_':
  97.                 attr = name.lower()
  98.                 self.assertEquals(getattr(result, attr), result[getattr(stat, name)])
  99.                 self.assert_(attr in members)
  100.                 continue
  101.         
  102.         
  103.         try:
  104.             result[200]
  105.             self.fail('No exception thrown')
  106.         except IndexError:
  107.             pass
  108.  
  109.         
  110.         try:
  111.             result.st_mode = 1
  112.             self.fail('No exception thrown')
  113.         except TypeError:
  114.             pass
  115.  
  116.         
  117.         try:
  118.             result.st_rdev = 1
  119.             self.fail('No exception thrown')
  120.         except (AttributeError, TypeError):
  121.             pass
  122.  
  123.         
  124.         try:
  125.             result.parrot = 1
  126.             self.fail('No exception thrown')
  127.         except AttributeError:
  128.             pass
  129.  
  130.         
  131.         try:
  132.             result2 = os.stat_result((10,))
  133.             self.fail('No exception thrown')
  134.         except TypeError:
  135.             pass
  136.  
  137.         
  138.         try:
  139.             result2 = os.stat_result((0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14))
  140.         except TypeError:
  141.             pass
  142.  
  143.  
  144.     
  145.     def test_statvfs_attributes(self):
  146.         if not hasattr(os, 'statvfs'):
  147.             return None
  148.         
  149.         import statvfs
  150.         
  151.         try:
  152.             result = os.statvfs(self.fname)
  153.         except OSError:
  154.             e = None
  155.             import errno
  156.             if e.errno == errno.ENOSYS:
  157.                 return None
  158.             
  159.         except:
  160.             e.errno == errno.ENOSYS
  161.  
  162.         self.assertEquals(result.f_bfree, result[statvfs.F_BFREE])
  163.         members = dir(result)
  164.         for name in dir(statvfs):
  165.             if name[:2] == 'F_':
  166.                 attr = name.lower()
  167.                 self.assertEquals(getattr(result, attr), result[getattr(statvfs, name)])
  168.                 self.assert_(attr in members)
  169.                 continue
  170.         
  171.         
  172.         try:
  173.             result.f_bfree = 1
  174.             self.fail('No exception thrown')
  175.         except TypeError:
  176.             pass
  177.  
  178.         
  179.         try:
  180.             result.parrot = 1
  181.             self.fail('No exception thrown')
  182.         except AttributeError:
  183.             pass
  184.  
  185.         
  186.         try:
  187.             result2 = os.statvfs_result((10,))
  188.             self.fail('No exception thrown')
  189.         except TypeError:
  190.             pass
  191.  
  192.         
  193.         try:
  194.             result2 = os.statvfs_result((0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14))
  195.         except TypeError:
  196.             pass
  197.  
  198.  
  199.  
  200. from test import mapping_tests
  201.  
  202. class EnvironTests(mapping_tests.BasicTestMappingProtocol):
  203.     '''check that os.environ object conform to mapping protocol'''
  204.     type2test = None
  205.     
  206.     def _reference(self):
  207.         return {
  208.             'KEY1': 'VALUE1',
  209.             'KEY2': 'VALUE2',
  210.             'KEY3': 'VALUE3' }
  211.  
  212.     
  213.     def _empty_mapping(self):
  214.         os.environ.clear()
  215.         return os.environ
  216.  
  217.     
  218.     def setUp(self):
  219.         self._EnvironTests__save = dict(os.environ)
  220.         os.environ.clear()
  221.  
  222.     
  223.     def tearDown(self):
  224.         os.environ.clear()
  225.         os.environ.update(self._EnvironTests__save)
  226.  
  227.  
  228.  
  229. class WalkTests(unittest.TestCase):
  230.     '''Tests for os.walk().'''
  231.     
  232.     def test_traversal(self):
  233.         import os
  234.         join = join
  235.         import os.path
  236.         sub1_path = join(test_support.TESTFN, 'SUB1')
  237.         sub11_path = join(sub1_path, 'SUB11')
  238.         sub2_path = join(test_support.TESTFN, 'SUB2')
  239.         tmp1_path = join(test_support.TESTFN, 'tmp1')
  240.         tmp2_path = join(sub1_path, 'tmp2')
  241.         tmp3_path = join(sub2_path, 'tmp3')
  242.         os.makedirs(sub11_path)
  243.         os.makedirs(sub2_path)
  244.         for path in (tmp1_path, tmp2_path, tmp3_path):
  245.             f = file(path, 'w')
  246.             f.write("I'm " + path + ' and proud of it.  Blame test_os.\n')
  247.             f.close()
  248.         
  249.         all = list(os.walk(test_support.TESTFN))
  250.         self.assertEqual(len(all), 4)
  251.         flipped = all[0][1][0] != 'SUB1'
  252.         all[0][1].sort()
  253.         self.assertEqual(all[0], (test_support.TESTFN, [
  254.             'SUB1',
  255.             'SUB2'], [
  256.             'tmp1']))
  257.         self.assertEqual(all[1 + flipped], (sub1_path, [
  258.             'SUB11'], [
  259.             'tmp2']))
  260.         self.assertEqual(all[2 + flipped], (sub11_path, [], []))
  261.         self.assertEqual(all[3 - 2 * flipped], (sub2_path, [], [
  262.             'tmp3']))
  263.         all = []
  264.         for root, dirs, files in os.walk(test_support.TESTFN):
  265.             all.append((root, dirs, files))
  266.             if 'SUB1' in dirs:
  267.                 dirs.remove('SUB1')
  268.                 continue
  269.         
  270.         self.assertEqual(len(all), 2)
  271.         self.assertEqual(all[0], (test_support.TESTFN, [
  272.             'SUB2'], [
  273.             'tmp1']))
  274.         self.assertEqual(all[1], (sub2_path, [], [
  275.             'tmp3']))
  276.         all = list(os.walk(test_support.TESTFN, topdown = False))
  277.         self.assertEqual(len(all), 4)
  278.         flipped = all[3][1][0] != 'SUB1'
  279.         all[3][1].sort()
  280.         self.assertEqual(all[3], (test_support.TESTFN, [
  281.             'SUB1',
  282.             'SUB2'], [
  283.             'tmp1']))
  284.         self.assertEqual(all[flipped], (sub11_path, [], []))
  285.         self.assertEqual(all[flipped + 1], (sub1_path, [
  286.             'SUB11'], [
  287.             'tmp2']))
  288.         self.assertEqual(all[2 - 2 * flipped], (sub2_path, [], [
  289.             'tmp3']))
  290.         for root, dirs, files in os.walk(test_support.TESTFN, topdown = False):
  291.             for name in files:
  292.                 os.remove(join(root, name))
  293.             
  294.             for name in dirs:
  295.                 os.rmdir(join(root, name))
  296.             
  297.         
  298.         os.rmdir(test_support.TESTFN)
  299.  
  300.  
  301.  
  302. class MakedirTests(unittest.TestCase):
  303.     
  304.     def setUp(self):
  305.         os.mkdir(test_support.TESTFN)
  306.  
  307.     
  308.     def test_makedir(self):
  309.         base = test_support.TESTFN
  310.         path = os.path.join(base, 'dir1', 'dir2', 'dir3')
  311.         os.makedirs(path)
  312.         path = os.path.join(base, 'dir1', 'dir2', 'dir3', 'dir4')
  313.         os.makedirs(path)
  314.         self.failUnlessRaises(OSError, os.makedirs, os.curdir)
  315.         path = os.path.join(base, 'dir1', 'dir2', 'dir3', 'dir4', 'dir5', os.curdir)
  316.         os.makedirs(path)
  317.         path = os.path.join(base, 'dir1', os.curdir, 'dir2', 'dir3', 'dir4', 'dir5', 'dir6')
  318.         os.makedirs(path)
  319.  
  320.     
  321.     def tearDown(self):
  322.         path = os.path.join(test_support.TESTFN, 'dir1', 'dir2', 'dir3', 'dir4', 'dir5', 'dir6')
  323.         while not os.path.exists(path) and path != test_support.TESTFN:
  324.             path = os.path.dirname(path)
  325.         os.removedirs(path)
  326.  
  327.  
  328.  
  329. class DevNullTests(unittest.TestCase):
  330.     
  331.     def test_devnull(self):
  332.         f = file(os.devnull, 'w')
  333.         f.write('hello')
  334.         f.close()
  335.         f = file(os.devnull, 'r')
  336.         self.assertEqual(f.read(), '')
  337.         f.close()
  338.  
  339.  
  340.  
  341. class URandomTests(unittest.TestCase):
  342.     
  343.     def test_urandom(self):
  344.         
  345.         try:
  346.             self.assertEqual(len(os.urandom(1)), 1)
  347.             self.assertEqual(len(os.urandom(10)), 10)
  348.             self.assertEqual(len(os.urandom(100)), 100)
  349.             self.assertEqual(len(os.urandom(1000)), 1000)
  350.         except NotImplementedError:
  351.             pass
  352.  
  353.  
  354.  
  355.  
  356. def test_main():
  357.     test_support.run_unittest(TemporaryFileTests, StatAttributeTests, EnvironTests, WalkTests, MakedirTests, DevNullTests, URandomTests)
  358.  
  359. if __name__ == '__main__':
  360.     test_main()
  361.  
  362.